這次來看看QLattice來執行迴歸的功能,
我們想從Airbnb上待租房子的特徵,來預測房子租金的價格。
參考攻略:
https://docs.abzu.ai/docs/tutorials/python/airbnb.html
首先先下載資料集:
上傳資料集到colab:
安裝載入套件:
去掉遺失的數值,查看資料結構:
切分訓練集測試集:
設定neighbourhood_group、neighbourhood、room_type這三個欄位為類別的特徵,
price為預測的標籤,開始訓練(迴歸少了kind參數):
利用圖形查看最佳的模型:
也可以利用待租房子經緯度查看真實租金與預測租金的差異: